package(default_visibility = ["//maga_transformer:__subpackages__"])

py_test (
    name = "all_fake_model_test",
    srcs = ["all_fake_model_test.py"],
    data = [
        "//maga_transformer/test/model_test/fake_test/testdata:testdata",
    ],
    deps = [
        "//maga_transformer:models",
        "//maga_transformer:_ft_pickler",
        "//maga_transformer:config",
        "//maga_transformer:testlib",
        "//maga_transformer/test/model_test/test_util:test_util",
    ],
)

py_test (
    name = "all_fake_safetensors_test",
    srcs = ["all_fake_safetensors_test.py"],
    data = [
        "//maga_transformer/test/model_test/fake_test/testdata:testdata",
    ],
    deps = [
        "//maga_transformer:models",
        "//maga_transformer:_ft_pickler",
        "//maga_transformer:config",
        "//maga_transformer:testlib",
        "//maga_transformer/test/model_test/test_util:test_util",
    ],
)